package scheme;

import java.util.HashMap;
import it.unisa.dia.gas.jpbc.Element;
import java.util.ArrayList;
import java.util.Map;

/**
 * @Classname CSP
 * @Description 云服务提供商
 * @Function 主要是保存信息，这里只有一个简单例子，实际保存的信息已使用.properties文件存储
 * @Created by kevin
 */
public class CSP {
    /**
     * 用来存储用户的属性集合，就是{S’_aid,uid}aid \in I_{AA}
     */
    public static Map< Element,ArrayList<Element>> listAtt = new HashMap<>();
    public static Map< Element,ArrayList<Element>> listAUK = new HashMap<>();

    /**
     * 保存用户盲化属性
     * @param Huid
     * @param TransAttrList
     */
    public static void saveUserTransAtt(Element Huid,Element[] TransAttrList){
        if (listAtt.containsKey(Huid)) {
            // 将Element[] b中的元素逐个添加到ArrayList中
            for (Element element : TransAttrList) {
                listAtt.get(Huid).add(element);
            }
        } else {
            // 如果不包含Element a，则创建一个新的ArrayList，并将Element[] b中的元素添加到其中
            ArrayList<Element> newList = new ArrayList<>();
            for (Element element : TransAttrList) {
                newList.add(element);
            }

            // 将Element a作为键，新的ArrayList作为值，插入到listAtt中
            listAtt.put(Huid,newList);
        }
    }
    /**
     * 保存用户属性认证密钥
     * @param Huid
     * @param TransAttrList
     */
    public static void saveUserAUK( Element Huid , Element[] TransAttrList){
        if (listAUK.containsKey(Huid)) {
            // 将Element[] b中的元素逐个添加到ArrayList中
            for (Element element : TransAttrList) {
                listAUK.get(Huid).add(element);
            }
        } else {
            // 如果不包含Element a，则创建一个新的ArrayList，并将Element[] b中的元素添加到其中
            ArrayList<Element> newList = new ArrayList<>();
            for (Element element : TransAttrList) {
                newList.add(element);
            }
            // 将Element a作为键，新的ArrayList作为值，插入到listAtt中
            listAUK.put(Huid,newList);
        }

    }

    /**
     * 查找对应盲化属性
     * @param Huid
     * @return
     */
    public static ArrayList<Element> findUserTransAtt(Element Huid) {
        // Initialize an ArrayList to store the result
        ArrayList<Element> resultList = new ArrayList<>();

        // Iterate over the entries of the map
        for (Map.Entry<Element, ArrayList<Element>> entry : listAtt.entrySet()) {
            // Get the key (H_uid) and value (ArrayList<Element>) of the current entry
            Element key = entry.getKey();
            ArrayList<Element> value = entry.getValue();

            // Check if the key matches the elementToFind
            if (key.equals(Huid)) {
                // If it matches, add the corresponding ArrayList<Element> to the resultList
                resultList.addAll(value);
                // Break the loop since we found the matching key
                break;
            }
        }
        // Return the resultList which may contain the ArrayList<Element> corresponding to the elementToFind
        return resultList;
    }
    /**
     * 查找对应盲化属性
     * @param Huid
     * @return
     */
    public static ArrayList<Element> findUserTransAuk(Element Huid) {
        // Initialize an ArrayList to store the result
        ArrayList<Element> resultList = new ArrayList<>();

        // Iterate over the entries of the map
        for (Map.Entry<Element, ArrayList<Element>> entry : listAUK.entrySet()) {
            // Get the key (H_uid) and value (ArrayList<Element>) of the current entry
            Element key = entry.getKey();
            ArrayList<Element> value = entry.getValue();

            // Check if the key matches the elementToFind
            if (key.equals(Huid)) {
                // If it matches, add the corresponding ArrayList<Element> to the resultList
                resultList.addAll(value);
                // Break the loop since we found the matching key
                break;
            }
        }
        // Return the resultList which may contain the ArrayList<Element> corresponding to the elementToFind
        return resultList;
    }

}
